package com.yoshino.leetcode.improve40.twentyseventh;

import java.util.ArrayList;
import java.util.List;

class Solution {

    List<List<Integer>> res = new ArrayList<>();
    List<Integer> temp = new ArrayList<>();
    public List<List<Integer>> subsets(int[] nums) {
        // 递归
        dfs(0, nums);
        return res;
    }

    private void dfs(int cur, int[] nums) {
        if (cur == nums.length) {
            // 不能直接使用 temp，不然是传的地址
            res.add(new ArrayList<>(temp));
            return;
        }
        temp.add(nums[cur]);
        dfs(cur + 1, nums);
        temp.remove(temp.size() - 1);
        // 跳过当前的
        dfs(cur + 1, nums);
    }


    public List<List<Integer>> combine(int n, int k) {
        // 递归
//        dfs(1, k, n + 1);
        dfs(1, k, n);
        return res;
    }

    private void dfs(int cur, int k, int n) {
//        if (cur > n) {
//            return;
//        }
        if (temp.size() + (n - cur + 1) < k) {
            return;
        }
        if (temp.size() == k) {
            res.add(new ArrayList<>(temp));
            return;
        }
        temp.add(cur);
//        dfs(cur + 1, k - 1, n);
        dfs(cur + 1, k, n);
        temp.remove(temp.size() - 1);
        dfs(cur + 1, k, n);

    }

    // 回溯
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        dfs(candidates, target, 0);
        return res;
    }

    private void dfs(int[] candidates, int target, int ind) {
        if (ind == candidates.length) {
            return;
        }
        if (target == 0) {
            res.add(new ArrayList<>(temp));
            return;
        }
        // 直接跳过
        dfs(candidates, target, ind + 1);
        // 选择当前数字
        if (target - candidates[ind] >= 0) {
            temp.add(candidates[ind]);
            dfs(candidates, target - candidates[ind], ind);
            temp.remove(temp.size() - 1);
        }
    }

}